package sjg.DataStructures.tree;

import sjg.DataStructures.util.SortUtils;

/**
 * 数组模拟二叉树，实现前中后序遍历
 * 父节点 = 1
 */
public class ArrBinaryTreeDemo {
    public static void main(String[] args) {
        int[] arr = {1, 2, 3, 4, 5, 6, 7};
        System.out.println("原数组遍历");
        SortUtils.print(arr);

        System.out.println("前序遍历");
        ArrBinaryTree arrBinaryTree = new ArrBinaryTree(arr);
        System.out.println("前序遍历，0为父节点");
        arrBinaryTree.preOrder();
        System.out.println("中序遍历，0为父节点");
        arrBinaryTree.infixOrder();
        System.out.println("后序遍历，0为父节点");
        arrBinaryTree.postOrder();

    }
}

// 数组模拟二叉树
class ArrBinaryTree {
    private int[] arr;// 存储数据节点的数组

    public ArrBinaryTree(int[] arr) {
        this.arr = arr;
    }

    // 重载 preOrder 前序
    public void preOrder() {
        this.preOrder(0);
    }

    // 重载 infixOrder 中序
    public void infixOrder() {
        this.infixOrder(0);
    }

    // 重载 postOrder 后序
    public void postOrder() {
        this.postOrder(0);
    }

    /**
     * 二叉树前序遍历
     */
    private void preOrder(int index) {
        // 如果数组为空，或者arr.length=0
        if (arr == null || arr.length == 0) {
            System.out.println("数组为空，不能按照二叉树的前序遍历");
        }
        // 输出当前这个元素
        System.out.println(arr[index]);
        // 向左递归遍历
        if ((index * 2 + 1) < arr.length) {
            preOrder(2 * index + 1);
        }
        // 向右递归遍历
        if ((index * 2 + 2) < arr.length) {
            preOrder(2 * index + 2);
        }
    }

    /**
     * 二叉树中序遍历
     */
    private void infixOrder(int index) {
        if (arr == null || arr.length == 0) {
            System.out.println("数组为空，不能按照二叉树的前序遍历");
        }
        // 向左递归遍历
        if ((index * 2 + 1) < arr.length) {
            infixOrder(2 * index + 1);
        }

        // 中
        System.out.println(arr[index]);

        // 向右递归遍历
        if ((index * 2 + 2) < arr.length) {
            infixOrder(2 * index + 2);
        }

    }

    /**
     * 二叉树后序遍历
     */
    private void postOrder(int index) {
        if (arr == null || arr.length == 0) {
            System.out.println("数组为空，不能按照二叉树的前序遍历");
        }
        // 向左递归遍历
        if ((index * 2 + 1) < arr.length) {
            postOrder(2 * index + 1);
        }
        // 向右递归遍历
        if ((index * 2 + 2) < arr.length) {
            postOrder(2 * index + 2);
        }
        // 中
        System.out.println(arr[index]);
    }
}
